package hot100.子串;

import java.util.HashMap;
import java.util.Map;

/**
 * 给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。
 * 子数组是数组中元素的连续非空序列。
 *
 * @ https://leetcode.cn/problems/subarray-sum-equals-k/?envType=study-plan-v2&envId=top-100-liked
 * @date 2025/08/24
 */
public class 和为K的子数组_560 {

    public static void main(String[] args) {

        和为K的子数组_560 impl = new 和为K的子数组_560();
        System.out.println(impl.subarraySum2(new int[]{1, 1, 1}, 2));

    }

    /**
     * 枚举
     */
    public int subarraySum(int[] nums, int k) {
        int cnt = 0;

        for (int i = 0; i < nums.length; i++) {
            int curSum = 0;
            for (int j = i; j < nums.length; j++) {
                curSum += nums[j];
                if (curSum == k) {
                    cnt++;
                }
            }
        }
        return cnt;
    }

    /**
     * 前缀和+哈希表
     * preSum[j] - pre[i] = k
     */
    public int subarraySum2(int[] nums, int k) {
        int cnt = 0;

        int[] preSum = new int[nums.length + 1];
        for (int i = 0; i < nums.length; i++) {
            preSum[i + 1] = preSum[i] + nums[i];
        }
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        for (int i = 1; i < preSum.length; i++) {
            int sub = preSum[i] - k;
            cnt += map.getOrDefault(sub, 0);
            map.put(preSum[i], map.getOrDefault(preSum[i], 0) + 1);
        }

        return cnt;
    }

}
